#pragma once

#include <iostream>
#include <cstdlib>
#include <signal.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/stat.h>
#define Root "/"            // 是否要更改的路径
#define devnull "/dev/null" // 重定向的目标文件,向里写入内容会被自动回收,从里面也读不出内容
void Daemon(bool ischdir, bool isclose)
{
    // 1.守护进程一般要屏蔽可能会引发异常的信号
    signal(SIGCHLD, SIG_IGN);
    signal(SIGPIPE, SIG_IGN);
    // 2.守护进程不能是进程组的组长,要成为非组长,可以fork后父进程直接退出,子进程继续执行(守护进程是特殊的孤儿进程)
    if (fork() > 0)
    {
        exit(0);
    }
    // 3.建立新会话,避免受当前用户登录注销的影响
    setsid();
    // 4.是否要更改CWD为根目录
    if (ischdir)
    {
        chdir(Root);
    }
    // 5.守护进程就不需要关心用户的输入输出和错误了,关闭和重定向由用户自己决定
    if (isclose)
    {
        close(0);
        close(1);
        close(2);
    }
    else
    {
        int fd = open(devnull, O_WRONLY);
        if (fd > 0)
        {
            dup2(fd, 0);
            dup2(fd, 1);
            dup2(fd, 2);
            close(fd);
        }
    }
}